package com.gitee.algorithm.linear;

import java.util.Iterator;

public class TwoWayLinkList<T> implements Iterable<T>{
    //首节点
    private Node head;
    //尾结点
    private Node last;
    //链表长度
    private int N;


    //结点类
    private class Node{
        public Node(T item,Node pre,Node next){
            this.item=item;
            this.pre=pre;
            this.next=next;
        }
        //储存数据
        public T item;
        //指向上一个结点
        public Node pre;
        //指向下一个结点
        public Node next;

    }

    //构造方法
    public TwoWayLinkList(){
        //初始化头结点和尾结点
        this.head=new Node(null,null,null);
        this.last=null;
        //初始化链表长度
        this.N=0;
    }
    //清空链表
    public void clear(){
        this.head.next=null;
        this.last.pre=null;
        this.N=0;
    }
    //获取链表长度
    public int length(){
        return N;
    }
    //判断链表是否为空
    public boolean isEmpty(){
        return N==0;
    }
    //获取第一个元素
    public T getFirst(){
        if(isEmpty()){
            return null;
        }
        return head.next.item;
    }
    //获取最后一个元素
    public T getLast(){
        if(isEmpty()){
            return null;
        }
        return last.item;
    }

    //插入元素
    public void insert(T t){
        //如果链表为空
        if(isEmpty()){
        //创建新节点
            Node newNode=new Node(t,head,null);
        //让新节点为尾结点
            last=newNode;
        //让头结点指向尾结点
            head.next=last;
            //如果不为空
        }else {
            Node oldNode=last;
            //创建新结点
            Node newNode = new Node(t, oldNode, null);
            //让当前的尾结点指向新结点
            oldNode.next=newNode;
            //让新结点成为尾结点
            last=newNode;
        }
        //元素个数加一
        N++;
    }

    //指定i位置插入元素
    public void insert(int i,T t){
        //找到i位置前一个结点
        Node pre=head;
        for(int index=0;index<i;index++){
            pre=pre.next;
        }
        //找到i位置结点
        Node curr=pre.next;
        //创建新结点
        Node newNode = new Node(t, pre, curr);
        //让i位置的前一个结点的下一个结点变成新结点
        pre.next=newNode;
        //让i位置的前一个结点变成新结点
        curr.pre=newNode;
        //元素个数加一
        N++;
    }

    //获取指定i位置的结点
    public T get(int i){
        Node n=head.next;
        for(int index=0;index<i;index++){
            n=n.next;
        }
        return n.item;
    }
    //查找到元素t第一次出现的位置
    public int indexOf(T t){
        Node n=head;
        for (int i=0;n.next!=null;i++){
            n=n.next;
            if(n.next.equals(t)){
                return i;
            }
        }
        return -1;
    }

    //指定i位置元素删除,并返回该元素
    public T remove(int i){
        //找到i位置前一个结点
        Node pre=head;
        for(int index=0;index<i;index++){
            pre=pre.next;
        }
        //找到i位置结点
        Node curr = pre.next;
        //找的i位置的下一个结点
        Node nextNode = curr.next;
        //让i位置的前一个结点的下一个结点变成i位置的下一个结点
        pre.next=nextNode;
        //让i位置的下一个结点的上一个结点变成i位置的前一个结点
        nextNode.pre=pre;
        //元素个数加一
        N--;
        return curr.item;
    }
//和迭代器相关
    @Override
    public Iterator<T> iterator() {
        return new TIterator();
    }
    private class TIterator implements Iterator{
        private Node n;
        public TIterator(){
           this.n=head;
        }
        @Override
        public boolean hasNext() {
            return n.next!=null;
        }

        @Override
        public Object next() {
            n=n.next;
            return n.item;
        }
    }
}
